package com.hongyi.common.web.interceptor;

import com.alibaba.fastjson2.JSON;
import com.hongyi.common.web.filter.RepeatedlyRequestWrapper;
import com.hongyi.common.framework.utils.common.MapUtils;
import com.hongyi.common.framework.utils.common.IoUtils;
import com.hongyi.common.framework.utils.spring.SpringUtils;
import com.hongyi.common.framework.utils.string.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.time.StopWatch;
import org.springframework.http.MediaType;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedReader;
import java.util.Map;

/**
 * web的调用时间统计拦截器
 * dev环境有效
 *
 * @author DingHao
 * @since 3.3.0
 */
@Slf4j
public class WebInvokeTimeInterceptor implements HandlerInterceptor {

    private final String prodProfile = "prod";

    /**
     * 创建一个线程隔离的变量ThreadLocal
     *
     *<p>ThreadLocal 是一个线程隔离的变量，每个线程都有自己独立的副本，互不影响</p>
     * **/
    private final ThreadLocal<StopWatch> invokeTl = new ThreadLocal<>();

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        if (!prodProfile.equals(SpringUtils.getActiveProfile())) {
            String url = request.getMethod()+" "+ request.getRequestURI();
            //如果请求的数据类型是json
            if (isJsonRequest(request)) {
                String jsonParam = "";
                //如果request 对象是否是一个RepeatedlyRequestWrapper 实例(RepeatedlyRequestWrapper继承自 HttpServletRequestWrapper，用于允许多次读取请求)
                if (request instanceof RepeatedlyRequestWrapper) {
                    //request 对象中获取一个 BufferedReader，用于读取请求体的内容
                    BufferedReader reader = request.getReader();
                    //Iog工具类读取BufferedReader中的所有内容，并将其转换为一个字符串
                    jsonParam = IoUtils.read(reader);
                }
                log.info("开始请求 => URL[{}],参数类型[json],参数:[{}]", url, jsonParam);
            } else {
                //从HttpServletRequest对象中获取所有请求参数的映射。getParameterMap 方法返回一个 Map，其中键是参数名，值是一个字符串数组，包含该参数的所有值
                Map<String, String[]> parameterMap = request.getParameterMap();
                if (MapUtils.isNotEmpty(parameterMap)) {
                    //将参数转为json字符串
                    String parameters = JSON.toJSONString(parameterMap);
                    log.info("开始请求 => URL[{}],参数类型[param],参数:[{}]", url, parameters);
                } else {
                    log.info("[PLUS]开始请求 => URL[{}],无参数", url);
                }
            }
            //创建一个用于测量时间的工具类对象
            StopWatch stopWatch = new StopWatch();
            //StopWatch 实例存储到 ThreadLocal 中
            invokeTl.set(stopWatch);
            //启动 StopWatch 的计时器
            stopWatch.start();
        }
        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        HandlerInterceptor.super.postHandle(request, response, handler, modelAndView);
    }

    /**
     * HandlerInterceptor 接口中的一个方法,请求处理完成后执行一些操作
     *
     * @param request 当前的 HttpServletRequest 对象
     * @param response 当前的 HttpServletResponse 对象
     * @param handler 处理请求的处理器对象
     * @param ex 如果请求处理过程中发生异常，这个参数会包含异常信息
     */
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        if (!prodProfile.equals(SpringUtils.getActiveProfile())) {
            //从 ThreadLocal 变量 invokeTl 中获取之前设置的 StopWatch 实例
            StopWatch stopWatch = invokeTl.get();
            //停止 StopWatch 的计时器
            stopWatch.stop();
            double timeInSeconds = stopWatch.getTime() / 1000.0; // 将毫秒转换为秒
            //记录请求的耗时信息
            log.info("结束请求 => URL[{}],耗时:[{}]秒", request.getMethod() + " " + request.getRequestURI(),timeInSeconds);
            //清除 ThreadLocal 变量
            invokeTl.remove();
        }
    }

    /**
     * 判断本次请求的数据类型是否为json
     *
     * @param request request
     * @return boolean
     */
    private boolean isJsonRequest(HttpServletRequest request) {
        //获取 Content-Type 头
        String contentType = request.getContentType();
        if (contentType != null) {
            //StringUtils.startsWithIgnoreCase 方法会忽略大小写地比较字符串的前缀。如果 Content-Type 以 application/json 开头，则返回 true，否则返回 false
            return StringUtils.startsWithIgnoreCase(contentType, MediaType.APPLICATION_JSON_VALUE);
        }
        return false;
    }

}
